package gla.centralhotel.dao;

import gla.centralhotel.dao.interfaces.IPriceDAO;
import gla.centralhotel.exception.CollectionRetrieverException;
import gla.centralhotel.exception.IllegalPriceException;
import gla.centralhotel.model.Price;
import gla.centralhotel.model.Room;
import java.util.ArrayList;
import java.util.List;
import javax.ejb.Stateless;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import javax.persistence.Query;

/**
 *
 * @author Rivo
 */
@Stateless
public class PriceDAO implements IPriceDAO {

    @PersistenceContext
    private EntityManager em;

    private Price create(Price price) {
        em.persist(price);
        return price;
    }

    private Price edit(Price price) {
        em.merge(price);
        return price;
    }

    public void remove(Price price) {
        em.remove(em.merge(price));
    }

    public Price find(Object id) {
        return em.find(gla.centralhotel.model.Price.class, id);
    }

    public List<Price> findAll() {
        return em.createQuery("select object(o) from Price as o").getResultList();
    }
    
    /**
     * Permet de recuperer une collection de manager d'hotel en passant en parametre
     * un room
     * @param room
     * @return
     * @throws gla.centralhotel.exception.CollectionRetrieverException
     */
    public ArrayList<Price> getAllPriceByRoom(Room room) throws CollectionRetrieverException{
        String query = "select price from Price as price where price.refRoom = :refRoom";
        Query q = em.createQuery(query);
        q.setParameter("refRoom", room);
        
        try{
            return new ArrayList(q.getResultList());
        }catch(IllegalStateException ex){
            ex.printStackTrace();
            throw new CollectionRetrieverException("Erreur de recuperation des prix par chambre");
        }
        
    }
    /**
     * Si un prix est affecté à une chambre sur une période, MAJ du prix
     * Sinon Ajouter prix
     * @param price
     * @return
     */
    public Price savePriceForRoom(Price price) throws IllegalPriceException{
        String periodVerif = "select price from Price as price where price.periodStart =" +
                ":start and price.periodEnd = :end and price.refRoom = :refRoom";
        Query q = em.createQuery(periodVerif);
        q.setParameter("start", price.getPeriodStart());
        q.setParameter("end", price.getPeriodEnd());
        q.setParameter("refRoom", price.getRefRoom());
        List<Price> priceList = q.getResultList();
        if (priceList.size() == 0) {
            return create(price);
        } else if(priceList.size() == 1){
            return edit(price);
        }else{
            throw new IllegalPriceException("Erreur grave il y a plusieurs prix " +
                    "sur une même période pour la même chambre");
        }
    }
}
